# encoding: utf-8

import lsptrain

from lsptrain.nlp_info_extract.train_scripts import train

train_path = "./train.txt"
test_path = "./test.txt"
pretrained_model = "uie-nano"
valid_steps = 2  # 测试多少步验证并保存训练权重，实际开发默认500
train(train_path=train_path, test_path=test_path, model=pretrained_model, logging_steps=valid_steps, valid_steps=valid_steps)
